server="4090_2"

assert server in ["172", "cluster", "autodl", "scm", "4090_1", "4090_2"]

if server == "4090_2":
    DATA_ROOT = "/data/qiaocongyu/data/ularef/"
    TINY_DATA_ROOT = "/data/qiaocongyu/data/ularef//tiny-imagenet-200/"
    PARTIAL_WEIGHT_ROOT = "/data/qiaocongyu/data/ularef/partial_weights/"
    REALWORLD_DATA_ROOT = "/data/qiaocongyu/data/ularef/PLL_realworld/"